import torch
import torch.distributions as dist

# 假设我们有一个正态分布
mean = torch.tensor([0.0, 0.0])
std = torch.tensor([1.0, 1.0])
distribution = dist.Normal(mean, std)

# 假设我们有一个动作
actions = torch.tensor([1.0, 2.0])

# 计算对数概率
log_probs = distribution.log_prob(actions)

# 对最后一个维度求和
log_prob_sum = log_probs.sum(dim=-1)

print(log_prob_sum)